import datetime, time
from Base.base import Base
from Page.mis.page_element import PageElements


class MisAudit(Base):

    def __init__(self):
        super().__init__("mis")
        # 当前月 -int类型数字
        self.cur_month = datetime.datetime.now().month
        # 获取当前日期+1天 - 字符串 '年-月-日'
        self.next_day = (datetime.datetime.now() + datetime.timedelta(days=+1)).strftime("%Y-%m-%d")

    def query_article(self, title):
        """
        查询需要审核的内容
        :param title: 查询内容
        :return:
        """
        # 输入文章标题
        self.input_ele(PageElements.audit_title, title)
        # 选择结束日期
        self.search_click(PageElements.audit_over_time)
        time.sleep(1)  # 等待弹框弹窗
        if self.cur_month == int(self.next_day.split('-')[1]):
            # 当前月 等于 +1天之后月 那么当前月份定位日期
            # 选择日期+1天
            new_loc = (PageElements.audit_add_day_before[0],
                       PageElements.audit_add_day_before[1].format(self.next_day.split('-')[2]))
        else:
            # 当前月 < +1天之后月 那么下个月份定位日期
            new_loc = (PageElements.audit_add_day_after[0],
                       PageElements.audit_add_day_after[1].format(self.next_day.split('-')[2]))

        self.search_click(new_loc)

        # 点击日期确定
        self.search_click(PageElements.audit_day_confirm_btn)
        # 点击查询
        self.search_click(PageElements.audit_btn_query)

    def article_pass(self):
        """文章审核通过"""
        time.sleep(2)  # 等待2S，等待页面更新数据
        # 点击通过按钮
        self.search_click(PageElements.audit_btn_pass)
        time.sleep(1)
        # 点击弹窗确定按钮
        self.search_click(PageElements.audit_pass_alert_btn)
